"""


Version: 0.1
Author: lk
Date: 2022-03-08 18:02
"""
import dgl
import torch
from dgl.nn.pytorch import Set2Set
from dgl.nn import pytorch

g1 = dgl.rand_graph(3, 4)
g1_node_feats = torch.randn(3, 5)
g2 = dgl.rand_graph(9, 8)
g2_node_feats = torch.randn(9, 5)

s2s = pytorch.Set2Set(5, 2, 1)

print(s2s(g1, g1_node_feats))

